package com.tedu.weibo.controller;

import com.tedu.weibo.common.result.JsonResult;
import com.tedu.weibo.common.result.StatusCode;
import com.tedu.weibo.mapper.UserMapper;
import com.tedu.weibo.pojo.dto.UserLoginDTO;
import com.tedu.weibo.pojo.dto.UserRegDTO;
import com.tedu.weibo.pojo.entity.User;
import com.tedu.weibo.pojo.vo.UserVO;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import springfox.documentation.annotations.ApiIgnore;

import javax.servlet.http.HttpSession;
import java.util.Date;

@Api(tags = "用户模块")
@RestController
@RequestMapping("/v1/users/")
public class UserController {
    @Autowired
    private UserMapper userMapper;
    @ApiOperation("用户注册功能")
    @PostMapping("reg")
    public JsonResult register(@RequestBody UserRegDTO userRegDTO){
        System.out.println(userRegDTO);

        String username = userRegDTO.getUsername();
        User findUser = userMapper.selectByUsername(username);
        if (findUser != null) {
            return new JsonResult(StatusCode.USERNAME_EXISTS);// 用户已存在
        }

        User user = new User();
        BeanUtils.copyProperties(userRegDTO, user);
        user.setCreated(new Date());
        userMapper.insert(user);
        return JsonResult.success(user);// 注册成功
    }

    @ApiOperation("用户登录功能")
    @PostMapping("login")
    public JsonResult login(@RequestBody @Validated UserLoginDTO userLoginDTO, @ApiIgnore HttpSession session) {
        System.out.println(userLoginDTO);
        /*
        if (userLoginDTO.getUsername() == null || userLoginDTO.getUsername().isEmpty()) {
        //  throw new RuntimeException("用户名不能为空！");
            throw new IllegalArgumentException("用户名不能为空！");
        }
        */

        User user = userMapper.selectByUsername(userLoginDTO.getUsername());
        if (user != null) {
            if (user.getPassword().equals(userLoginDTO.getPassword())) {
                session.setAttribute("user", user);
                return JsonResult.success();// 登录成功
            }else {
                return new JsonResult(StatusCode.PASSWORD_ERROR);// 密码错误
            }
        }else {
            return new JsonResult(StatusCode.USERNAME_ERROR);// 用户不存在
        }
    }

    @ApiOperation("获取当前登录用户信息功能")
    @GetMapping("currentUser")
    public JsonResult currentUser(@ApiIgnore HttpSession session) {
        User user = (User) session.getAttribute("user");
        if (user == null) {
            return new JsonResult(StatusCode.NOT_LOGIN);// 未登录
        }else {
            UserVO userVO = new UserVO();
            BeanUtils.copyProperties(user, userVO);
            return JsonResult.success(userVO);
        }
    }

    @ApiOperation("用户登出功能")
    @GetMapping("logout")
    public void logout(@ApiIgnore HttpSession session) {
        session.removeAttribute("user");
    }
}
